import numpy as np
import matplotlib.pyplot as plt
import sonar
plt.rcParams['font.sans-serif'] = ['SimHei']  
# Matplotlib中设置字体-黑体，解决Matplotlib中文乱码问题
plt.rcParams['axes.unicode_minus'] = False    
# 解决Matplotlib坐标轴负号'-'显示为方块的问题
fs=20e3 #采样率
T=1/fs; #采样周期
L=fs
t=np.arange(L)/fs

#线谱信号
f0=1000
theta_d=30*np.pi/180
SNR=10
s=np.exp(1j*(2*np.pi*f0*t))
ti=np.arange(100)
theta=np.zeros(ti.shape)
theta_f=np.zeros(ti.shape)
for i in range(ti.size):
    X=sonar.vec_signal(s,SNR,theta_d,3)
    theta[i]=sonar.sointen_t(X)*180/np.pi
    
    theta_f[i]=sonar.sointen_f(X,fs)*180/np.pi

plt.plot(ti,theta_f)


plt.ylim(29,31)
plt.xlabel('实验次数')
plt.ylabel('估计角度')
plt.title('平均声强器')
plt.show()